#!/usr/bin/env bash

# ------------------------------------------------------------------------------
# Function to clone or update a repository.
# Arguments:
#   1. repo_path: URL of the repository to clone.
#   2. repo_branch: Branch of the repository to checkout.
# Usage:
#   clone_repo <repo_path> <repo_branch>
# ------------------------------------------------------------------------------
clone_repo() {
    local repo_path="$1"
    local repo_branch="$2"
    local repo_basename repo_username repo_name
    local repo_info="${repo_path#*:}"

    repo_username="$(basename "$(dirname "${repo_info}")")"
    repo_basename=$(basename "${repo_info}" .git)
    repo_name="${repo_basename}_${repo_username}"

    echo "Cloning ${repo_name}..." >&2
    if [ ! -d "${repo_name}" ]; then
        git clone --recursive "${repo_path}" "${repo_name}" >&2
    elif [ -d "${repo_name}/.git" ]; then
        git -C "${repo_name}" fetch --all >&2
    else
        echo "... Warning: ${repo_name} is not a git repository." >&2
        echo "... clone failed." >&2
        echo "... git pull failed." >&2
        echo "... Please remove ${repo_name} and try again." >&2
        return 1
    fi

    git -C "${repo_name}" checkout "${repo_branch}" >&2
    git -C "${repo_name}" show --oneline -s >&2

    echo "${repo_name}"
}


repo_name=$(clone_repo "https://github.com/danpetreamd/MAD.git" "aac_tweaks")

pushd "${repo_name}"

echo
echo "installing prerequisites..."
pip install -r requirements.txt
echo

echo
echo "verifying installation..."
madengine run --tags pyt_huggingface_bert
echo

popd